tensor([[1.0000, 0.1200, 1.0000, 0.0000, 1.0000],
        [0.0000, 0.8800, 0.0000, 1.0000, 0.0000]], grad_fn=<RoundBackward1>)